#!/usr/bin/env python
#-*- coding:utf-8 -*- 

from numpy import *

l = diag((0, 1, 2, 3, 4))
l = mat(l)
print(l)

